python average function for beginners